package com.example.service;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.example.common.Constants;
import com.example.common.enums.ErrorCodeEnum;
import com.example.common.PageResult;
import com.example.common.enums.RoleEnum;
import com.example.exception.CustomException;
import com.example.mapper.AdminMapper;
import com.example.pojo.entity.Account;
import com.example.pojo.entity.Admin;
import com.example.utils.AccountUtils;
import com.example.utils.TokenUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.ArrayList;
import java.util.List;

/**
 * 管理员 service 层实现类
 */
@Service
public class AdminServiceImpl extends ServiceImpl<AdminMapper, Admin> implements AdminService {

    /**
     * 创建一个管理员账号
     *
     * @param admin 管理员账号信息
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void add(Admin admin) {
        // 1.判空
        if (admin == null) {
            throw new CustomException(ErrorCodeEnum.PARAM_ERROR);
        }
        // 2.用户名不能为空
        if (StrUtil.isBlank(admin.getUsername())) {
            throw new CustomException(ErrorCodeEnum.USER_NAME_IS_NOT_NULL);
        }
        // 3.没有昵称默认用用户名
        if (StrUtil.isBlank(admin.getName())) {
            admin.setName(admin.getUsername());
        }
        // 4.没有密码默认用默认密码
        if (StrUtil.isBlank(admin.getPassword())) {
            admin.setPassword(Constants.USER_DEFAULT_PASSWORD);
        }
        // 5.角色是管理员
        admin.setRole(RoleEnum.ADMIN.name());
        // 6.判断此用户名是否存在
        Admin dbAdmin = this.lambdaQuery()
                .eq(Admin::getUsername, admin.getUsername())
                .one();
        if (dbAdmin != null) {
            throw new CustomException(ErrorCodeEnum.USER_EXIST_ERROR);
        }

        // 7.创建账号
        this.save(admin);
    }

    /**
     * 分页查询账号
     *
     * @param pageNum  页码
     * @param pageSize 页大小
     * @param admin 条件
     * @return 分页结果
     */
    @Override
    public PageResult<Admin> pageQuery(Integer pageNum, Integer pageSize, Admin admin) {
        // 1.校验参数
        if (pageNum == null || pageNum < 1) {
            throw new CustomException(ErrorCodeEnum.PAGE_QUERY_NUM);
        }
        if (pageSize == null || pageSize < 1 || pageSize > 100) {
            throw new CustomException(ErrorCodeEnum.PAGE_QUERY_SIZE);
        }

        // 2.创建 MybatisPlus 分页对象
        IPage<Admin> pageObject = new Page<>(pageNum, pageSize);

        // 3.分页查询账号
        pageObject = this.getBaseMapper().pageQuery(pageObject, admin);

        // 4.构建分页结果并返回
        return new PageResult<>(
                pageObject.getTotal(),
                pageObject.getRecords()
        );
    }

    /**
     * 修改账号
     *
     * @param admin 账号信息
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void update(Admin admin) {
        this.updateById(admin);
    }

    /**
     * 删除账号
     *
     * @param id 账号id
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void delete(Integer id) {
        // 1.校验参数
        if (id == null) {
            throw new CustomException(ErrorCodeEnum.PARAM_ERROR);
        }
        // 2.判断要删除的管理员账号是否是当前登录的管理员
        List<Integer> ids = new ArrayList<>();
        ids.add(id);
        this.checkCurrentUser(ids);
        // 3.根据账号 id 删除账号
        this.removeById(id);
    }

    /**
     * 批量删除账号
     *
     * @param ids 账号id列表
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteBatch(List<Integer> ids) {
        // 1.校验参数
        if(CollectionUtil.isEmpty(ids)) {
            throw new CustomException(ErrorCodeEnum.PARAM_ERROR);
        }
        // 2.判断要删除的管理员账号是否包含当前登录的管理员
        this.checkCurrentUser(ids);
        // 3.根据账号 id 列表批量删除账号
        this.removeByIds(ids);
    }

    /**
     * 判断要删除的管理员账号是否包含当前登录的管理员
     *
     * @param ids 要删除的管理员账号 id
     */
    private void checkCurrentUser(List<Integer> ids) {
        // 1.获取当前登录的管理员
        Account currentLoginAccount = AccountUtils.getCurrentLoginAccount();
        if (currentLoginAccount == null) {
            throw new CustomException(ErrorCodeEnum.GET_CUR_USER_ERROR);
        }
        Admin dbAdmin = this.lambdaQuery()
                .eq(Admin::getUsername, currentLoginAccount.getUsername())
                .one();
        if (dbAdmin == null) {
            throw new CustomException(ErrorCodeEnum.GET_CUR_USER_ERROR);
        }
        // 2.判断要删除的账号是否包含当前登录的管理员
        if (ids.contains(dbAdmin.getId())) {
            throw new CustomException(ErrorCodeEnum.NOT_DELETE_CURRENT_ADMIN);
        }
    }

    /**
     * 登录
     *
     * @param account 账号
     * @return 账号
     */
    @Override
    public Admin login(Account account) {
        // 1.校验账号是否存在
        Admin dbAdmin = this.lambdaQuery()
                .eq(Admin::getUsername, account.getUsername())
                .one();
        if (dbAdmin == null) {
            throw new CustomException(ErrorCodeEnum.USER_NOT_EXIST_ERROR);
        }
        // 2.校验密码
        if (!dbAdmin.getPassword().equals(account.getPassword())) {
            throw new CustomException(ErrorCodeEnum.USER_PASSWORD_ERROR);
        }

        // 3.创建 jwt token
        // token 内容 = 账号id-角色，如 23-ADMIN
        // 密钥 = 账号密码
        String token = TokenUtils.createToken(dbAdmin.getId() + "-" + dbAdmin.getRole(), dbAdmin.getPassword());
        // 4.封装账号
        // 4.1 token存入对象
        dbAdmin.setToken(token);
        // 4.2 不返回密码
        dbAdmin.setPassword(null);
        // 5.返回账号
        return dbAdmin;
    }

    /**
     * 修改账号密码
     *
     * @param account 账号
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updatePassword(Account account) {
        // 1.根据用户名称校验账号是否存在
        Admin dbAdmin = this.lambdaQuery()
                .eq(Admin::getUsername, account.getUsername())
                .one();
        if (dbAdmin == null) {
            throw new CustomException(ErrorCodeEnum.USER_NOT_EXIST_ERROR);
        }
        // 2.根据用户输入的原密码校验与数据库的密码是否一致
        if (!dbAdmin.getPassword().equals(account.getPassword())) {
            throw new CustomException(ErrorCodeEnum.PARAM_OLD_PASSWORD_ERROR);
        }
        // 3.将用户输入的新密码存入数据库
        this.lambdaUpdate()
                .eq(Admin::getUsername, account.getUsername())
                .set(Admin::getPassword, account.getNewPassword())
                .update();
    }
}
